how to fix while loop